from .utils import BaseXlsxWrite


class TestWriter(BaseXlsxWrite):
    def create_xlsx_filename(self):
        return 'test.xlsx'

    def set_hearde(self, key=None):
        """
        pt = {
            'id': None,     # 单元格坐标ID
            'x': None,      # X轴位置
            'y': None,      # Y轴位置
            'value': '',    # 单元格的值
            'bg_color': '',  # 单元格背景颜色
            'is_edit': 1,   # 是否可以编辑，前端展示使用
            'x_use': 1,     # X轴所占格数
            'y_use': 1     # Y轴所占格数
        }
        如果 key：是一个列表关键字，手动根据关键字设置
        """
        head_list = [
            {'id': '1_1', 'x': 1, 'y': 1, 'value': '时间大标题', 'bg_color': '', 'is_edit': 1, 'x_use': 6, 'y_use': 1, 'is_heade': 1},
            {'id': '1_2', 'x': 1, 'y': 2, 'value': '日期', 'bg_color': '', 'is_edit': 1, 'x_use': 1, 'y_use': 1, 'is_heade': 1},
            {'id': '2_2', 'x': 2, 'y': 2, 'value': '电', 'bg_color': '', 'is_edit': 1, 'x_use': 3, 'y_use': 1, 'is_heade': 1},
            {'id': '5_2', 'x': 5, 'y': 2, 'value': '区域', 'bg_color': '', 'is_edit': 1, 'x_use': 1, 'y_use': 1, 'is_heade': 1},
            {'id': '6_2', 'x': 6, 'y': 2, 'value': '价格', 'bg_color': '', 'is_edit': 1, 'x_use': 1, 'y_use': 1, 'is_heade': 1},
            {'id': '1_3', 'x': 1, 'y': 3, 'value': 'val_1', 'bg_color': '', 'is_edit': 1, 'x_use': 1, 'y_use': 1},
            {'id': '2_3', 'x': 2, 'y': 3, 'value': 'val_2', 'bg_color': '', 'is_edit': 1, 'x_use': 1, 'y_use': 1},
            {'id': '3_3', 'x': 3, 'y': 3, 'value': 'val_3', 'bg_color': '', 'is_edit': 1, 'x_use': 1, 'y_use': 1},
            {'id': '4_3', 'x': 4, 'y': 3, 'value': 'val_4', 'bg_color': '', 'is_edit': 1, 'x_use': 1, 'y_use': 1},
            {'id': '5_3', 'x': 5, 'y': 3, 'value': 'val_5', 'bg_color': '', 'is_edit': 1, 'x_use': 1, 'y_use': 1},
            {'id': '6_3', 'x': 6, 'y': 3, 'value': 'val_6', 'bg_color': '', 'is_edit': 1, 'x_use': 1, 'y_use': 1}
        ]
        self.hearde = head_list

    def get_data(self):
        """
        ['id', 'x', 'y', 'value', 'x_use', 'y_use']
        """
        pt = [
            {'id': '1_4', 'x': 1, 'y': 4, 'value': '一', 'bg_color': '', 'is_edit': 1, 'x_use': 1, 'y_use': 1},
            {'id': '2_4', 'x': 2, 'y': 4, 'value': '二', 'bg_color': '', 'is_edit': 1, 'x_use': 3, 'y_use': 1}
        ]
        if self.hearde:
            self.data = self.hearde + pt
        else:
            self.data = pt
